grid challenge python solution